from owl_admin.ext import db


class LLMModelPO(db.Model):
    __tablename__ = 'chat_model'
    __table_args__ = {'comment': '聊天模型'}

    id = db.Column(db.Integer, primary_key=True)
    model_name = db.Column(db.String(100), nullable=False)
    api_type = db.Column(db.String(255), nullable=False)
    api_host = db.Column(db.String(255), nullable=False)
    api_key = db.Column(db.String(255), nullable=False)
    model_describe: db.Column = db.Column(db.String(255), nullable=False)
    vision_support = db.Column(db.Boolean, nullable=False)
    audio_support = db.Column(db.Boolean, nullable=False)
    create_by = db.Column(db.Integer, nullable=False)
    create_time = db.Column(db.DateTime, nullable=False)
    update_by = db.Column(db.Integer, nullable=False)
    update_time = db.Column(db.DateTime, nullable=False)
